// Copyright 2018 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef V8_ALLOCATION_SITE_SCOPES_INL_H_
#define V8_ALLOCATION_SITE_SCOPES_INL_H_

#include "src/allocation-site-scopes.h"

#include "src/objects/allocation-site-inl.h"

namespace v8 {
namespace internal {

    void AllocationSiteContext::InitializeTraversal(Handle<AllocationSite> site)
    {
        top_ = site;
        // {current_} is updated in place to not create unnecessary Handles, hence
        // we initially need a separate handle.
        current_ = Handle<AllocationSite>::New(*top_, isolate());
    }

    Handle<AllocationSite> AllocationSiteUsageContext::EnterNewScope()
    {
        if (top().is_null()) {
            InitializeTraversal(top_site_);
        } else {
            // Advance current site
            Object nested_site = current()->nested_site();
            // Something is wrong if we advance to the end of the list here.
            update_current_site(AllocationSite::cast(nested_site));
        }
        return Handle<AllocationSite>(*current(), isolate());
    }

    void AllocationSiteUsageContext::ExitScope(Handle<AllocationSite> scope_site,
        Handle<JSObject> object)
    {
        // This assert ensures that we are pointing at the right sub-object in a
        // recursive walk of a nested literal.
        DCHECK(object.is_null() || *object == scope_site->boilerplate());
    }

    bool AllocationSiteUsageContext::ShouldCreateMemento(Handle<JSObject> object)
    {
        if (activated_ && AllocationSite::CanTrack(object->map()->instance_type())) {
            if (FLAG_allocation_site_pretenuring || AllocationSite::ShouldTrack(object->GetElementsKind())) {
                if (FLAG_trace_creation_allocation_sites) {
                    PrintF("*** Creating Memento for %s %p\n",
                        object->IsJSArray() ? "JSArray" : "JSObject",
                        reinterpret_cast<void*>(object->ptr()));
                }
                return true;
            }
        }
        return false;
    }

} // namespace internal
} // namespace v8

#endif // V8_ALLOCATION_SITE_SCOPES_INL_H_
